Sine graph in python